package zisu.rabbitmqspringboot.rbsb.producer;

import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component;
import zisu.rabbitmqspringboot.rbsb.entity.Order;

import java.util.Map;

/**
 * @Author: 蔡翔
 * @Date: 2019/8/16 10:33
 * @Version 1.0
 */
@Component
public class RabbitSender {
    @Autowired
    private RabbitTemplate rabbitTemplate;

    // ConfirmCallback用于监听 Broker端给我们返回的确认请求。
    //无论消息可不可达，都会执行ConfirmCallback这个方法。
    final RabbitTemplate.ConfirmCallback confirmCallback = new RabbitTemplate.ConfirmCallback() {
        @Override
        public void confirm(CorrelationData correlationData, boolean ack, String cause) {
            System.err.println("correlationData:"+correlationData);
            System.err.println("ack:"+ack);
            if(!ack){
                System.err.println("异常处理。。。");
            }
        }
    };

    //如果消息不可达，routingkey不匹配，那么需要做二次处理，如补偿处理。
    //如果消息可达就不会到这一步。只有在消息不可达，才会执行这个方法。
    final RabbitTemplate.ReturnCallback returnCallback = new RabbitTemplate.ReturnCallback() {
        @Override
        public void returnedMessage(org.springframework.amqp.core.Message message, int replyCode, String replyText, String exchange, String routingKey) {
            System.err.println("return exchange:"+exchange+",routingKey:"+routingKey+",replyCode"+replyCode+",replyText："+replyText);
        }
    };

    // 生产端 ，消息的生产。
    public void send(Object message, Map<String,Object> properties) throws Exception{
        MessageHeaders mhs = new MessageHeaders(properties);
        Message message1 = MessageBuilder.createMessage(message,mhs);

        rabbitTemplate.setConfirmCallback(confirmCallback);  //设置消息的监听请求 回送
        rabbitTemplate.setReturnCallback(returnCallback);
        CorrelationData cd = new CorrelationData();  //如果这里不设置CorrelationData，那么我们监听到消息的CorrelationData就为null
        cd.setId("1234567");        //id + 时间戳   注意要全局唯一，这个就是实际消息的id，以后再进行消息可靠性投递的时候，就要通过这个id找到唯一的一条消息。

        rabbitTemplate.convertAndSend("exchange-1","springboot.hello",message1,cd);
    }

    // 生产端 ，消息的生产。
    public void sendOrder(Order order, Map<String,Object> properties) throws Exception{
        MessageHeaders mhs = new MessageHeaders(properties);
        Message message1 = MessageBuilder.createMessage(order,mhs);

        rabbitTemplate.setConfirmCallback(confirmCallback);  //设置消息的监听请求 回送
        rabbitTemplate.setReturnCallback(returnCallback);
        CorrelationData cd = new CorrelationData();  //如果这里不设置CorrelationData，那么我们监听到消息的CorrelationData就为null
        cd.setId("1234567a");        //id + 时间戳   注意要全局唯一，这个就是实际消息的id，以后再进行消息可靠性投递的时候 ，就要通过这个id找到唯一的一条消息。

        rabbitTemplate.convertAndSend("exchange-2","AGV1.def",message1,cd);
    }
    public void sendOrder2(Order order, Map<String,Object> properties) throws Exception{
        MessageHeaders mhs = new MessageHeaders(properties);
        Message message1 = MessageBuilder.createMessage(order,mhs);

        rabbitTemplate.setConfirmCallback(confirmCallback);  //设置消息的监听请求 回送
        rabbitTemplate.setReturnCallback(returnCallback);
        CorrelationData cd = new CorrelationData();  //如果这里不设置CorrelationData，那么我们监听到消息的CorrelationData就为null
        cd.setId("1234567a");        //id + 时间戳   注意要全局唯一，这个就是实际消息的id，以后再进行消息可靠性投递的时候 ，就要通过这个id找到唯一的一条消息。

        rabbitTemplate.convertAndSend("exchange-2","AGV2.def",message1,cd);
    }
}
